import * as React from "react"
import { cva, type VariantProps } from "class-variance-authority"
import { cn } from "@/lib/utils"

const buttonVariants = cva(
  "inline-flex items-center justify-center gap-2 whitespace-nowrap rounded-md text-sm font-medium transition-all disabled:pointer-events-none disabled:opacity-50 [&_svg]:pointer-events-none [&_svg:not([class*='size-'])]:size-4 shrink-0 [&_svg]:shrink-0 outline-none focus-visible:border-ring focus-visible:ring-ring/50 focus-visible:ring-[3px] aria-invalid:ring-destructive/20 dark:aria-invalid:ring-destructive/40 aria-invalid:border-destructive",
  {
    variants: {
      variant: {
        default: "bg-primary text-primary-foreground shadow-xs hover:bg-primary/90",
        destructive: "bg-destructive text-white shadow-xs hover:bg-destructive/90 focus-visible:ring-destructive/20 dark:focus-visible:ring-destructive/40 dark:bg-destructive/60",
        outline: "border bg-background shadow-xs hover:bg-accent hover:text-accent-foreground dark:bg-input/30 dark:border-input dark:hover:bg-input/50",
        secondary: "bg-secondary text-secondary-foreground shadow-xs hover:bg-secondary/80",
        ghost: "hover:bg-accent hover:text-accent-foreground dark:hover:bg-accent/50",
        link: "text-primary underline-offset-4 hover:underline",
      },
      size: {
        default: "h-9 px-4 py-2 has-[>svg]:px-3",
        sm: "h-8 rounded-md gap-1.5 px-3 has-[>svg]:px-2.5",
        lg: "h-10 rounded-md px-6 has-[>svg]:px-4",
        icon: "size-9",
      },
    },
    defaultVariants: {
      variant: "default",
      size: "default",
    },
  }
)

type CustomVariant = 'primary' | 'secondary' | 'ghost' | 'outline' | 'destructive' | 'link'

export interface ButtonProps
  extends React.ButtonHTMLAttributes<HTMLButtonElement> {
  variant?: CustomVariant
  size?: 'default' | 'sm' | 'lg' | 'icon'
}

const Button = React.forwardRef<HTMLButtonElement, ButtonProps>(
  ({ className, variant = 'primary', size = 'default', ...props }, ref) => {
    let colorClass = ''
    switch (variant) {
      case 'primary':
        colorClass = 'bg-wechat text-white font-semibold rounded-lg active:bg-wechat-dark transition-colors hover:bg-wechat/90 border-none'
        break
      case 'secondary':
        colorClass = 'bg-gray-100 text-gray-900 font-semibold rounded-lg hover:bg-gray-200 border border-gray-200'
        break
      case 'ghost':
        colorClass = 'bg-transparent text-wechat font-semibold rounded-lg hover:bg-wechat/10 active:bg-wechat/20'
        break
      case 'outline':
        colorClass = 'border border-wechat text-wechat font-semibold rounded-lg bg-white hover:bg-wechat/10 active:bg-wechat/20'
        break
      case 'destructive':
        colorClass = 'bg-red-500 text-white font-semibold rounded-lg hover:bg-red-600'
        break
      case 'link':
        colorClass = 'bg-transparent text-wechat underline hover:text-wechat-dark'
        break
    }

    return (
      <button
        ref={ref}
        className={cn(
          'px-4 py-2 text-base transition-colors disabled:opacity-50 disabled:cursor-not-allowed',
          colorClass,
          className
        )}
        {...props}
      />
    )
  }
)

Button.displayName = 'Button'

export { Button, buttonVariants }
